package 简单.滑动窗口;

import java.util.*;

/**
 * 给你一个整数数组 nums 和一个整数 k ，判断数组中是否存在两个 不同的索引 i 和 j ，
 * 满足 nums[i] == nums[j] 且 abs(i - j) <= k 。如果存在，返回 true ；否则，返回 false 。
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/contains-duplicate-ii
 */
public class 存在重复元素II_219 {

    public static void main(String[] args) {

        int[] a = {1, 2, 3, 1};
        boolean b = containsNearbyDuplicate2(a, 3);
        System.out.println(b);

    }

    //超时
    public static boolean containsNearbyDuplicate1(int[] nums, int k) {
        List<Integer> list = new ArrayList<>();
        for (int num : nums) {
            if (list.contains(num)) {
                return true;
            }
            list.add(num);
            if (list.size() > k) {
                list.remove(0);
            }
        }
        return false;
    }

    /**
     * 滑动窗口+Map存储元素个数
     *
     * @param nums
     * @param k
     * @return
     */
    public static boolean containsNearbyDuplicate2(int[] nums, int k) {
        k = k + 1;  // abs(i - j) <= k ，i和j是索引，其实窗口元素个数，最大为k+1
        Map<Integer, Integer> map = new HashMap<>();
        int length = nums.length;
        for (int i = 0; i < length && i < k; i++) {  //先将开始k个元素存入map
            if (map.getOrDefault(nums[i], 0) == 1) {
                return true;
            }
            map.put(nums[i], 1);
        }
        for (int i = k; i < length; i++) {
            map.put(nums[i - k], 0);  //窗口向后滑动一个位置
            if (map.getOrDefault(nums[i], 0) == 1) {
                return true;
            }
            map.put(nums[i], 1);
        }
        return false;
    }

    /**
     * 滑动窗口+利用set自动去重
     */
    public static boolean efficientContainsNearbyDuplicate(int[] nums, int k) {
        HashSet<Integer> set = new HashSet<>();
        for (int i = 0; i < nums.length; i++) {
            if (!set.add(nums[i])) {
                return true;
            }
            if (set.size() > k) {
                //每次遍历一个元素则将其加入哈希表中，如果哈希表的大小大于 k，则移除最前面的数字
                set.remove(nums[i - k]);
            }
        }
        return false;
    }

}
